package com.larry.mr.partitioner2;

import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapreduce.Partitioner;

/**
 * @ProjectName hadoop
 * @Author larry
 * @Date 10:43 2021/11/16
 * @Description 自定义分区
 **/
public class PhonePartitioner extends Partitioner<Text,FlowBean> {
    @Override
    public int getPartition(Text text, FlowBean flowBean, int numPartitions) {

        int par;
        String phone = text.toString();

        if (phone.startsWith("136")){
            par = 0;
        }else if (phone.startsWith("137")){
            par = 1;
        }else if (phone.startsWith("138")){
            par = 2;
        }else if (phone.startsWith("139")){
            par = 3;
        }else {
            par = 4;
        }

        return par;
    }
}
